import torch
import timeit

#初始化张量
# x = torch.tensor([[1],[2]])
# print(x.shape)
# print(x.dtype)
#
# print(x * 10)
# print(x.add(10))
# print(x.view(-1))
# print(x.squeeze(1))
# print(torch.cat([x,x],axis=1))
x = torch.rand(1,6400).to('cuda')
y = torch.rand(6400,5000).to('cuda')
timeit_cpu = timeit.timeit("x@y",globals = globals(),number=100000)
print(timeit_cpu)